from AdminFunc import Admin
flag=1

#如果flag为真，则一直进入循环
while flag:
    #初始化一个admin对象
    a=Admin()
    res=input("您是否为首次登录系统:如首次登录需要设置用户名密码")
    if res=="yes":
        #先创建必要文件
        txt=["AdminUsernameAndpassword.txt","StudentUsernameAndpassword.txt","TeacherUsernameAndpassword.txt"]
        for i in txt:
            with open(i, 'w') as file:
                print("创建{}文件成功".format(i))

        #设置管理员
        a.admin_set()
    elif res=="no":
        admin_account=input("请输入管理员账号")
        admin_passoword=input("请输入管理员密码")
        if admin_account==a.show_admin()[0] and admin_passoword==a.show_admin()[1]:
            print("1、重置管理员密码")
            print("2、增加学生")
            print("3、删除学生")
            print("4、查询学生列表")
            print("5、增加老师")
            print("6、删除老师")
            print("7、查询老师列表")
            print("8、退出系统")
            while True:
                key_word=input("输入数字即可使用对应功能:")
                if key_word=="1":
                    a.admin_set()
                elif key_word=="2":
                    a.add_student()
                elif key_word=="3":
                    a.delete_student()
                elif key_word=="4":
                    a.show_all_student()
                elif key_word=="5":
                    a.add_teacher()
                elif key_word=="6":
                    a.delete_teacher()
                elif key_word=="7":
                    a.show_all_teacher()
                elif key_word=="8":
                     flag=0
                     break
        else:
            print("登录失败，请检查用户名密码是否正确")
            break
    elif res=="q":
        print("已退出管理员系统")
        break
